/* USER CODE BEGIN Header */
/**
  ******************************************************************************
  * @file           : usbd_storage_if.c
  * @version        : v1.0_Cube
  * @brief          : Memory management layer.
  ******************************************************************************
  * @attention
  *
  * Copyright (c) 2025 STMicroelectronics.
  * All rights reserved.
  *
  * This software is licensed under terms that can be found in the LICENSE file
  * in the root directory of this software component.
  * If no LICENSE file comes with this software, it is provided AS-IS.
  *
  ******************************************************************************
  */
/* USER CODE END Header */

/* Includes ------------------------------------------------------------------*/
#include "usbd_storage_if.h"

/* USER CODE BEGIN INCLUDE */

/* USER CODE END INCLUDE */

/* Private typedef -----------------------------------------------------------*/
/* Private define ------------------------------------------------------------*/
/* Private macro -------------------------------------------------------------*/

/* USER CODE BEGIN PV */
/* Private variables ---------------------------------------------------------*/
HAL_StatusTypeDef res;

extern SD_HandleTypeDef hsd;

extern HAL_SD_CardInfoTypeDef cardInfo;

#include "diskio.h"

extern DRESULT SD_read(BYTE lun, BYTE *buff, DWORD sector, UINT count);

int click;

int Is_process_read = 0;
int read_finiehed = 0;

int Is_process_write = 0;
int write_finished = 0;

uint8_t *sector_buf;
uint32_t sector_addr;
uint16_t sector_size;

/* USER CODE END PV */

/** @addtogroup STM32_USB_OTG_DEVICE_LIBRARY
  * @brief Usb device.
  * @{
  */

/** @defgroup USBD_STORAGE
  * @brief Usb mass storage device module
  * @{
  */

/** @defgroup USBD_STORAGE_Private_TypesDefinitions
  * @brief Private types.
  * @{
  */

/* USER CODE BEGIN PRIVATE_TYPES */

/* USER CODE END PRIVATE_TYPES */

/**
  * @}
  */

/** @defgroup USBD_STORAGE_Private_Defines
  * @brief Private defines.
  * @{
  */

#define STORAGE_LUN_NBR                  1
#define STORAGE_BLK_NBR                  0x10000
#define STORAGE_BLK_SIZ                  0x200

/* USER CODE BEGIN PRIVATE_DEFINES */

/* USER CODE END PRIVATE_DEFINES */

/**
  * @}
  */

/** @defgroup USBD_STORAGE_Private_Macros
  * @brief Private macros.
  * @{
  */

/* USER CODE BEGIN PRIVATE_MACRO */

/* USER CODE END PRIVATE_MACRO */

/**
  * @}
  */

/** @defgroup USBD_STORAGE_Private_Variables
  * @brief Private variables.
  * @{
  */

/* USER CODE BEGIN INQUIRY_DATA_HS */
/** USB Mass storage Standard Inquiry Data. */
const int8_t STORAGE_Inquirydata_HS[] = {/* 36 */

	/* LUN 0 */
	0x00,
	0x80,
	0x02,
	0x02,
	(STANDARD_INQUIRY_DATA_LEN - 5),
	0x00,
	0x00,
	0x00,
	'S', 'T', 'M', ' ', ' ', ' ', ' ', ' ', /* Manufacturer : 8 bytes */
	'P', 'r', 'o', 'd', 'u', 'c', 't', ' ', /* Product      : 16 Bytes */
	' ', ' ', ' ', ' ', ' ', ' ', ' ', ' ',
	'0', '.', '0', '1'                      /* Version      : 4 Bytes */
};
/* USER CODE END INQUIRY_DATA_HS */

/* USER CODE BEGIN PRIVATE_VARIABLES */

/* USER CODE END PRIVATE_VARIABLES */

/**
  * @}
  */

/** @defgroup USBD_STORAGE_Exported_Variables
  * @brief Public variables.
  * @{
  */

extern USBD_HandleTypeDef hUsbDeviceHS;

/* USER CODE BEGIN EXPORTED_VARIABLES */

/* USER CODE END EXPORTED_VARIABLES */

/**
  * @}
  */

/** @defgroup USBD_STORAGE_Private_FunctionPrototypes
  * @brief Private functions declaration.
  * @{
  */

static int8_t STORAGE_Init_HS(uint8_t lun);
static int8_t STORAGE_GetCapacity_HS(uint8_t lun, uint32_t *block_num, uint16_t *block_size);
static int8_t STORAGE_IsReady_HS(uint8_t lun);
static int8_t STORAGE_IsWriteProtected_HS(uint8_t lun);
static int8_t STORAGE_Read_HS(uint8_t lun, uint8_t *buf, uint32_t blk_addr, uint16_t blk_len);
static int8_t STORAGE_Write_HS(uint8_t lun, uint8_t *buf, uint32_t blk_addr, uint16_t blk_len);
static int8_t STORAGE_GetMaxLun_HS(void);

/* USER CODE BEGIN PRIVATE_FUNCTIONS_DECLARATION */

/* USER CODE END PRIVATE_FUNCTIONS_DECLARATION */

/**
  * @}
  */

USBD_StorageTypeDef USBD_Storage_Interface_fops_HS =
{
  STORAGE_Init_HS,
  STORAGE_GetCapacity_HS,
  STORAGE_IsReady_HS,
  STORAGE_IsWriteProtected_HS,
  STORAGE_Read_HS,
  STORAGE_Write_HS,
  STORAGE_GetMaxLun_HS,
  (int8_t *)STORAGE_Inquirydata_HS
};

/* Private functions ---------------------------------------------------------*/

/**
  * @brief  Initializes the storage unit (medium).
  * @param  lun: Logical unit number.
  * @retval USBD_OK if all operations are OK else USBD_FAIL
  */
int8_t STORAGE_Init_HS(uint8_t lun)
{
  /* USER CODE BEGIN 9 */
	UNUSED(lun);

	printf("STORAGE_Init_HS \r\n");

	return (USBD_OK);
  /* USER CODE END 9 */
}

/**
  * @brief  Returns the medium capacity.
  * @param  lun: Logical unit number.
  * @param  block_num: Number of total block number.
  * @param  block_size: Block size.
  * @retval USBD_OK if all operations are OK else USBD_FAIL
  */
int8_t STORAGE_GetCapacity_HS(uint8_t lun, uint32_t *block_num, uint16_t *block_size)
{
  /* USER CODE BEGIN 10 */
	UNUSED(lun);

	// *block_num  = STORAGE_BLK_NBR;
	// *block_size = STORAGE_BLK_SIZ;
	// printf("STORAGE_GetCapacity_HS \r\n");

	*block_num  = cardInfo.LogBlockNbr;
	*block_size = cardInfo.BlockSize;

	return (USBD_OK);
  /* USER CODE END 10 */
}

/**
  * @brief   Checks whether the medium is ready.
  * @param  lun:  Logical unit number.
  * @retval USBD_OK if all operations are OK else USBD_FAIL
  */
int8_t STORAGE_IsReady_HS(uint8_t lun)
{
  /* USER CODE BEGIN 11 */
	UNUSED(lun);

	// printf("STORAGE_IsReady_HS \r\n");

	return (USBD_OK);
  /* USER CODE END 11 */
}

/**
  * @brief  Checks whether the medium is write protected.
  * @param  lun: Logical unit number.
  * @retval USBD_OK if all operations are OK else USBD_FAIL
  */
int8_t STORAGE_IsWriteProtected_HS(uint8_t lun)
{
  /* USER CODE BEGIN 12 */
	// printf("STORAGE_IsWriteProtected_HS \r\n");

	return (USBD_OK);
  /* USER CODE END 12 */
}

/**
  * @brief  Reads data from the medium.
  * @param  lun: Logical unit number.
  * @param  buf: data buffer.
  * @param  blk_addr: Logical block address.
  * @param  blk_len: Blocks number.
  * @retval USBD_OK if all operations are OK else USBD_FAIL
  */
int8_t STORAGE_Read_HS(uint8_t lun, uint8_t *buf, uint32_t blk_addr, uint16_t blk_len)
{
  /* USER CODE BEGIN 13 */
	// UNUSED(lun);
	// UNUSED(buf);
	// UNUSED(blk_addr);
	// UNUSED(blk_len);



	// if(disk_read(0, buf, blk_addr, blk_len) != 0) {
	// 	printf("STORAGE_Read_FS error -> %u  %d\r\n", blk_addr, blk_len);
	// }


	res = HAL_SD_ReadBlocks_DMA(&hsd, buf, blk_addr, blk_len);
	// res = HAL_SD_ReadBlocks(&hsd, buf, blk_addr, blk_len, 10240);
	if(res != HAL_OK) {
		printf("STORAGE_Read_HS error -> %u  %d  <> ret : 0x%x\r\n", blk_addr, blk_len, res);
		return (USBD_FAIL);
	}
	
	while(HAL_SD_GetState(&hsd) == HAL_SD_STATE_BUSY){};
	while( HAL_SD_GetCardState(&hsd) != HAL_SD_CARD_TRANSFER ) {}


	// SD_read(0, buf, blk_addr, blk_len);

	/***
	printf("STORAGE_Read_HS  addr = %d len = %d\r\n", blk_addr, blk_len);

	sector_buf = buf;
	sector_addr = blk_addr;
	sector_size = blk_len;

	read_finiehed = 0;
	Is_process_read = 1;

	click = 0;
	while((read_finiehed ==0) && (click < 180000000)) {

		if(read_finiehed == 1) break;



		click++;
		if(click > 180000000) {
			printf("STORAGE_Read_HS timeout !\r\n");
			return (USBD_FAIL);
		}

		if( HAL_SD_GetCardState(&hsd) == HAL_SD_CARD_TRANSFER ) {
			if(read_finiehed == 1)  break;
		}

	}
	***/

	return (USBD_OK);
  /* USER CODE END 13 */
}

/**
  * @brief  Writes data into the medium.
  * @param  lun: Logical unit number.
  * @param  buf: data buffer.
  * @param  blk_addr: Logical block address.
  * @param  blk_len: Blocks number.
  * @retval USBD_OK if all operations are OK else USBD_FAIL
  */
int8_t STORAGE_Write_HS(uint8_t lun, uint8_t *buf, uint32_t blk_addr, uint16_t blk_len)
{
  /* USER CODE BEGIN 14 */
	// UNUSED(lun);
	// UNUSED(buf);
	// UNUSED(blk_addr);
	// UNUSED(blk_len);

	// printf("STORAGE_Write_HS addr = %d len = %d\r\n", blk_addr, blk_len);

	// if(disk_write(0, buf, blk_addr, blk_len) != 0) {
	// 	printf("STORAGE_Write_FS error -> %u  %d\r\n", blk_addr, blk_len);
	// }
	
	// res = HAL_SD_WriteBlocks(&hsd, buf, blk_addr, blk_len, 10240);
	res = HAL_SD_WriteBlocks_DMA(&hsd, buf, blk_addr, blk_len);
	if(res != HAL_OK) {
		printf("STORAGE_Write_HS error -> %u  %d  <> ret : 0x%x\r\n", blk_addr, blk_len, res);
		return (USBD_FAIL);
	}
	
	while(HAL_SD_GetState(&hsd) == HAL_SD_STATE_BUSY){};
	while( HAL_SD_GetCardState(&hsd) != HAL_SD_CARD_TRANSFER ) {}

	return (USBD_OK);
  /* USER CODE END 14 */
}

/**
  * @brief  Returns the Max Supported LUNs.
  * @param  None
  * @retval Lun(s) number.
  */
int8_t STORAGE_GetMaxLun_HS(void)
{
  /* USER CODE BEGIN 15 */
	// printf("STORAGE_GetMaxLun_HS \r\n");

	return (STORAGE_LUN_NBR - 1);
  /* USER CODE END 15 */
}

/* USER CODE BEGIN PRIVATE_FUNCTIONS_IMPLEMENTATION */

/* USER CODE END PRIVATE_FUNCTIONS_IMPLEMENTATION */

/**
  * @}
  */

/**
  * @}
  */

